# coding: utf-8
"""
@Time    : 2024/8/28 9:47
@Author  : Y.H LEE
"""

from utils.tools import *
# from minibatch_iterator import MinibatchIterator
from grid_search import GridSearch
from sys_params import *
from benchmarks.TSP import TravelSalesPerson

'''
Model main function
'''


def main():
    """set Random seed"""
    set_seed(123456)

    """load params and grid search params"""
    hp_params = load_json(h_params)
    sc_params = load_json(search_params)

    """Questions"""
    env = TravelSalesPerson(hp_params["N_CITIES"], interactive=False)

    """model train"""
    GridSearch(hp_params, sc_params, env).train_with_hyperparams_grid_search(save_model=False)
    # trainer(minibatch)


if __name__ == '__main__':
    main()
